Not All Disagreement Is Learnable: Token Teachability in On-Policy Distillation
策略蒸馏中的令牌可教性:并非所有分歧都可学习
🧭 LLM 训练 · 策略蒸馏令牌选择优化
The Hong Kong Polytechnic University · InfiX.ai
提出令牌可教性区分可学习与不可兼容的分歧,TA-OPD 仅保留 5% 高可教性令牌蒸馏即可超越全令牌训练
📌 工作参考价值(4 点)
- 蒸馏效率与效果提升:当在电商场景中蒸馏LLM用于商品描述、推荐理由或Agent对话时,可借鉴TA-OPD仅对“可教性”高的token位置施加蒸馏,避免低效甚至有害的监督信号,实验仅用5%令牌即超越全令牌蒸馏,显著节省计算。
- 无需额外模型:TA-OPD仅利用教师和学生logits计算局部兼容性,无需奖励模型或验证器,易于集成到现有蒸馏流程(如生成式推荐中的语义ID生成器蒸馏)。
- 信号质量重新审视:揭示了原始KL散度作为学习价值代理的缺陷,提示在设计蒸馏损失时,应优先关注教师分布与学生top-K候选重叠高的token,而非简单加权高分歧位置,这对多智能体策略迁移中避免灾难性遗忘有参考价值。
- 算法轻量化:可教性计算简单,能在训练时动态筛选token,为推荐系统或Agent中频繁更新的小模型提供了一种低成本、高性能的在线蒸馏策略。